package com.xjj.datarources.base;

/**
 * @name: xjj
 * @date: 2021/5/18 13:42
 */
public class DBContextHandler {
    // 对当前线程的操作-线程安全的
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    // 调用此方法，切换数据源
    public static void setDataSource(String dataSource) {
        contextHolder.set(dataSource);
        System.out.println("已经切换到数据源:"+contextHolder.get());
    }

    // 获取数据源
    public static String getDataSource() {
        return contextHolder.get();
    }

    // 删除数据源
    public static void clearDataSource() {
        contextHolder.remove();
    }

}
